tensor([[0.6800, 0.6800, 0.6800, 0.6800, 0.6800],
        [0.3200, 0.3200, 0.3200, 0.3200, 0.3200]], grad_fn=<RoundBackward1>)